import xlrd
from xlrd import xldate_as_tuple
import xlwt

import datetime
# xlrd中单元格的数据类型
# 数字一律按浮点型输出，日期输出成一串小数，布尔型输出0或1，所以我们必须在程序中做判断处理转换
# 成我们想要的数据类型
# 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error

class ExcelData():
    # 初始化方法
    def __init__(self, data_path, sheetname):
        #定义一个属性接收文件路径
        self.data_path = data_path
        # 定义一个属性接收工作表名称
        self.sheetname = sheetname
        # 使用xlrd模块打开excel表读取数据
        self.data = xlrd.open_workbook(self.data_path)
        # 根据工作表的名称获取工作表中的内容（方式①）
        self.table = self.data.sheet_by_name(self.sheetname)
        # 根据工作表的索引获取工作表的内容（方式②）
        # self.table = self.data.sheet_by_name(0)
        # 获取第一行所有内容,如果括号中1就是第二行，这点跟列表索引类似
        self.keys = self.table.row_values(0)
        # 获取工作表的有效行数
        self.rowNum = self.table.nrows
        # 获取工作表的有效列数
        self.colNum = self.table.ncols

    # 定义一个读取excel表的方法
    def readExcel(self):
        # 定义一个空列表
        datas = []
        for i in range(1, self.rowNum):
            # 定义一个空字典
            sheet_data = {}
            for j in range(self.colNum):
                # 获取单元格数据类型
                c_type = self.table.cell(i,j).ctype
                # 获取单元格数据
                c_cell = self.table.cell_value(i, j)
                if c_type == 2 and c_cell % 1 == 0:  # 如果是整形
                    c_cell = int(c_cell)
                elif c_type == 3:
                    # 转成datetime对象
                    date = datetime.datetime(*xldate_as_tuple(c_cell,0))
                    c_cell = date.strftime('%Y/%d/%m %H:%M:%S')
                elif c_type == 4:
                    c_cell = True if c_cell == 1 else False
                sheet_data[self.keys[j]] = c_cell
                # 循环每一个有效的单元格，将字段与值对应存储到字典中
                # 字典的key就是excel表中每列第一行的字段
                # sheet_data[self.keys[j]] = self.table.row_values(i)[j]
            # 再将字典追加到列表中
            datas.append(sheet_data)
        # 返回从excel中获取到的数据：以列表存字典的形式返回
        return datas  
    
def getaverage(data):
    
        for i in range(15):
            workbook = xlwt.Workbook(encoding = 'utf-8')
            worksheet = workbook.add_sheet('Sheet1')
            titel=['s_ID','year','month','tem','rain','log','lat']
            # 生成标题
            for x in range(7):
                worksheet.write(0,x,titel[x])
            # 开始读取各年份数据并写入新表格
            for j in range(10):
               index=i*10+j
            #    print(data[index])
            #    循环读取数据对象中的value值
               for z in range(7):
                   dickey=titel[z]
                   worksheet.write(j+1,z, data[index][dickey])
            name='C:\\Users\\xulzu\\Desktop\\paper\\'+str(2005+i) +'data.xls'
            workbook.save(name) # 保存
        

 
        # 求出每年每月的气温和降水均值    
def get_year_average():
    datapath="C:\\Users\\xulzu\\Desktop\\paperdata\\origin_data.xlsx"
    # 遍历六张表
    workbook = xlwt.Workbook(encoding = 'utf-8')
    worksheet = workbook.add_sheet('Sheet1')
    worksheet2 = workbook.add_sheet('Sheet2')

    for month in range(6):
        sheetname='Sheet'+str(month+1)
        get_data=ExcelData(datapath,sheetname)
        datas=get_data.readExcel()
        # 遍历十五年
        for year in range(15):
            # 遍历每一年
            sumtem=0
            sumrain=0
            for everyyear in range(10):
                index=year*10+everyyear
                sumtem+=datas[index]['tem']
                sumrain+=datas[index]['rain']
            worksheet.write(month,year,sumtem/10)
            worksheet2.write(month,year,sumrain/10)  
    workbook.save("junzhi.xls") # 保存
if __name__ == "__main__":
    # data_path = "C:\\Users\\xulzu\\Desktop\\temAndRain\\origin_data.xlsx"
    # sheetname = "qiyue"
    # get_data = ExcelData(data_path, sheetname)
    # datas = get_data.readExcel()
    # getaverage(datas)
    get_year_average()
    


    
    # print(datas)

 
 